// Copyright 2014 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CC_TEST_MOCK_OCCLUSION_TRACKER_H_
#define CC_TEST_MOCK_OCCLUSION_TRACKER_H_

#include "base/macros.h"
#include "cc/trees/occlusion_tracker.h"

namespace cc {

class MockOcclusionTracker : public OcclusionTracker {
    // This class is used for testing only. It fakes just enough information to
    // calculate unoccluded content rect and unoccluded contributing surface
    // content rect. It calls the helper function of occlusion tracker to avoid
    // DCHECKs since testing environment won't be set up properly to pass those.
public:
    MockOcclusionTracker()
        : OcclusionTracker(gfx::Rect(0, 0, 1000, 1000))
    {
        OcclusionTracker::StackObject stack_obj;
        OcclusionTracker::stack_.push_back(stack_obj);
        OcclusionTracker::stack_.push_back(stack_obj);
    }

    explicit MockOcclusionTracker(const gfx::Rect& screen_scissor_rect)
        : OcclusionTracker(screen_scissor_rect)
    {
        OcclusionTracker::StackObject stack_obj;
        OcclusionTracker::stack_.push_back(stack_obj);
        OcclusionTracker::stack_.push_back(stack_obj);
    }

    void set_occluded_target_rect(const gfx::Rect& occluded)
    {
        OcclusionTracker::stack_.back().occlusion_from_inside_target = occluded;
    }

    void set_occluded_target_rect_for_contributing_surface(
        const gfx::Rect& occluded)
    {
        OcclusionTracker::stack_[OcclusionTracker::stack_.size() - 2]
            .occlusion_from_inside_target
            = occluded;
    }

private:
    DISALLOW_COPY_AND_ASSIGN(MockOcclusionTracker);
};

} // namespace cc

#endif // CC_TEST_MOCK_OCCLUSION_TRACKER_H_
